import { useNavigate, useLocation } from 'react-router-dom';

const Login = () => {
  const navigate = useNavigate();
  const location = useLocation();
  const from = location.state?.from || '/'; // 记录跳转前的路径

  // 模拟登录（实际项目中替换为真实接口请求）
  const handleLogin = (role) => {
    localStorage.setItem('user', JSON.stringify({
      id: 1,
      name: 'Current User',
      role // 角色：admin/editor/viewer
    }));
    navigate(from, { replace: true }); // 登录后跳回原路径
  };

  return (
    <div style={{ padding: '2rem' }}>
      <h1>Login Page</h1>
      <button onClick={() => handleLogin('editor')} style={{ marginRight: '10px' }}>
        登录为编辑者（editor）
      </button>
      <button onClick={() => handleLogin('admin')}>
        登录为管理员（admin）
      </button>
    </div>
  );
};

export default Login;